Multi-Stream Attention Residuals
多流注意力残差
🧭 高效 LLM 预训练 · 深度残差架构
School of AI, Shanghai Jiao Tong University · Shanghai Innovation Institute · Dots Studio, Xiaohongshu Inc. · School of CS, Peking University
将单流深度注意力残差扩展为多流深度注意力,通过多尺度因果卷积生成多样化上下文视图,在 MoE 大模型上取得 1.58 倍预训练加速
📌 工作参考价值(4 点)
- 若业务需要自研或微调大模型作为排序/召回/Agent 底座,可评估用 MSAR 替换标准残差连接:保持单流存储、按需扩展多流深度检索,iso-loss 预训练加速 1.58×,对 MoE 架构尤其明显。
- 多尺度因果卷积生成不同上下文视图的思路可迁移到用户行为序列建模:使用不同 kernel size 的因果卷积提取短期点击、中期会话、长期兴趣等多时间尺度表示,再通过自适应混合融合,增强序列建模能力。
- 轻量自适应混合多个检索流的机制,可作为推荐模型中多兴趣/多视图融合的替代方案:避免显式维护多流表示和通信开销,降低工程实现复杂度。
- 需注意该工作验证于 18B/28B 大规模 MoE 语言模型预训练,直接搬到电商推荐的小模型或非 Transformer 结构收益未知;引入多流机制可能增加调试复杂度,建议先在离线评测中验证深度特征稀释问题是否真实存在。